#!/bin/bash

docker run -it --rm \
  --gpus '"device=0"' \
  --ipc=host \
  --ulimit memlock=-1 \
  --ulimit stack=67108864 \
  -v /workspace/Qwen2-0.5B:/workspace/Qwen2-0.5B \
  -v /root/test-sts/test.py:/workspace/test.py \
  -v /root/test-sts/test-dist.py:/workspace/test-dist.py \
  test-sts:v2 \
  /bin/bash

